//\file: Image.h
//Modified from Image.h provided from lab1
#ifndef __ATOMTRACE_IMAGE_H_
#define __ATOMTRACE_IMAGE_H_

#include "Color.h"
#include "trax.hpp"

// Helper class to handle writes to the frame buffer

namespace AtomTrace{

    class Image {
    public:
        Image(int xres, int yres, int start_fb);

        void set(int i, int j, const Color& c);
        void set(int pix, const Color & c);
        
        Color get(int i, int j) const;
        Color get(int pix) const;

     protected:
        int xres, yres, start_fb;
    };

} // namespace AtomTrace

#endif //__ATOMTRACE_IMAGE_H_

